Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

C++ layer to implement Triu #3631

Merged
merged 11 commits into from
Jan 6, 2025
Merged

C++ layer to implement Triu #3631

merged 11 commits into from
Jan 6, 2025

Conversation

protonu
Copy link
Collaborator

@protonu protonu commented Dec 21, 2024

This is a C++ API to implement triu. The PR (#3637) stacked on top of this used it to create a Python interface.

Another way of using this may be to use the components of triu such as iota, broadcast, le and where from Thunder directly bypassing the need for a C++ implementation. As future work this commit/PR may be removed in favor of a Thunder only implementation.

@protonu protonu force-pushed the pbasu_iota_experiment branch from 260141f to 5f82c65 Compare December 23, 2024 21:33
@protonu protonu force-pushed the pbasu_iota_experiment branch from 0b42b74 to b734bf6 Compare January 2, 2025 20:12
@protonu
Copy link
Collaborator Author

protonu commented Jan 2, 2025

!test

@protonu
Copy link
Collaborator Author

protonu commented Jan 2, 2025

!test

@protonu protonu changed the title [WIP] C++ layer to implement Triu C++ layer to implement Triu Jan 2, 2025
@protonu protonu requested a review from jacobhinkle January 2, 2025 20:57
@protonu protonu marked this pull request as ready for review January 2, 2025 20:57
@protonu protonu requested a review from jacobhinkle January 2, 2025 21:20
@protonu
Copy link
Collaborator Author

protonu commented Jan 2, 2025

!test

@protonu protonu requested a review from jacobhinkle January 3, 2025 16:52
@protonu
Copy link
Collaborator Author

protonu commented Jan 4, 2025

!test

@protonu protonu requested a review from jacobhinkle January 4, 2025 15:23
@protonu
Copy link
Collaborator Author

protonu commented Jan 4, 2025

!test

Copy link
Collaborator

@jacobhinkle jacobhinkle left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@protonu
Copy link
Collaborator Author

protonu commented Jan 6, 2025

!test

@protonu protonu merged commit 1f18cf7 into main Jan 6, 2025
48 checks passed
@protonu protonu deleted the pbasu_iota_experiment branch January 6, 2025 19:51
protonu added a commit that referenced this pull request Jan 8, 2025
This PR exposes the Triu C++ API which was added in the PR
(#3631)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants